<template>
  <div>
    <el-upload
      :action="uploadUrl"
      :before-upload="handleBeforeUpload"
      :on-success="handleUploadSuccess"
      :on-error="handleUploadError"
      name="file"
      :show-file-list="false"
      :headers="headers"
      style="display: none"
      ref="upload"
      v-if="this.type == 'url'"
    >
    </el-upload>
    <div class="editor" ref="editor" :style="styles"></div>
  </div>
</template>

<script>
import Quill from "quill";
import "quill/dist/quill.core.css";
import "quill/dist/quill.snow.css";
import "quill/dist/quill.bubble.css";
import { getToken } from "@/utils/auth";
import BlotFormatter from "quill-blot-formatter"; // 新增导入

// 在组件顶部添加
const Image = Quill.import("formats/image");

class CustomImageBlot extends Image {
  static create(value) {
    let node = super.create(value);
    if (typeof value === "object") {
      // 处理对齐方式
      if (value.align === "left") {
        node.style.float = "left";
        node.style.margin = "0 1em 1em 0";
      } else if (value.align === "right") {
        node.style.float = "right";
        node.style.margin = "0 0 1em 1em";
      } else if (value.align === "center") {
        node.style.display = "block";
        node.style.margin = "auto";
      }

      // 确保URL正确
      if (value.url) {
        node.setAttribute("src", value.url);
      }
    }
    return node;
  }

  static value(node) {
    // 获取当前对齐状态
    let align = null;
    if (node.style.float === "left") {
      align = "left";
    } else if (node.style.float === "right") {
      align = "right";
    } else if (node.style.display === "block" && node.style.margin === "auto") {
      align = "center";
    }

    return {
      url: node.getAttribute("src"),
      align: align,
    };
  }
}

export default {
  name: "Editor",
  props: {
    /* 编辑器的内容 */
    value: {
      type: String,
      default: "",
    },
    /* 高度 */
    height: {
      type: Number,
      default: null,
    },
    /* 最小高度 */
    minHeight: {
      type: Number,
      default: null,
    },
    /* 只读 */
    readOnly: {
      type: Boolean,
      default: false,
    },
    // 上传文件大小限制(MB)
    fileSize: {
      type: Number,
      default: 100,
    },
    /* 类型（base64格式、url格式） */
    type: {
      type: String,
      default: "url",
    },
  },
  data() {
    return {
      uploadUrl: process.env.VUE_APP_BASE_API + "/system/oss/upload", // 上传的图片服务器地址
      headers: {
        Authorization: "Bearer " + getToken(),
      },
      Quill: null,
      currentValue: "",
      options: {
        theme: "snow",
        bounds: document.body,
        debug: "warn",
        modules: {
          // 工具栏配置
          toolbar: [
            ["bold", "italic", "underline", "strike"], // 加粗 斜体 下划线 删除线
            ["blockquote", "code-block"], // 引用  代码块
            [{ list: "ordered" }, { list: "bullet" }], // 有序、无序列表
            [{ indent: "-1" }, { indent: "+1" }], // 缩进
            [{ size: ["small", false, "large", "huge"] }], // 字体大小
            [{ header: [1, 2, 3, 4, 5, 6, false] }], // 标题
            [{ color: [] }, { background: [] }], // 字体颜色、字体背景颜色
            [{ align: [] }], // 对齐方式
            ["clean"], // 清除文本格式
            ["link", "image", "video"], // 链接、图片、视频
          ],
          // 新增 blotFormatter 配置
          blotFormatter: {
            blotTypes: ["image", "video"], // 支持调整图片和视频
            resize: {
              enabled: true,
              handleStyle: {
                backgroundColor: "black",
                border: "none",
              },
            },
            align: {
              enabled: true,
              options: ["left", "center", "right"],
            },
            drag: {
              enabled: true,
            },
          },
        },
        placeholder: "请输入内容",
        readOnly: this.readOnly,
        videoIndex: 0,
      },
    };
  },
  computed: {
    styles() {
      let style = {};
      if (this.minHeight) {
        style.minHeight = `${this.minHeight}px`;
      }
      if (this.height) {
        style.height = `${this.height}px`;
      }
      return style;
    },
  },
  watch: {
    value: {
      handler(val) {
        if (val !== this.currentValue) {
          this.currentValue = val || "";
          if (this.Quill) {
            this.Quill.pasteHTML(this.currentValue);
          }
        }
      },
      immediate: true,
    },
  },
  mounted() {
    this.init();
  },
  beforeDestroy() {
    this.Quill = null;
  },
  methods: {
    init() {
      // 注册 blotFormatter 模块
      Quill.register("modules/blotFormatter", BlotFormatter);
      Quill.register("formats/image", CustomImageBlot);

      const editor = this.$refs.editor;
      this.Quill = new Quill(editor, this.options);
      // 如果设置了上传地址则自定义图片上传事件
      if (this.type == "url") {
        let toolbar = this.Quill.getModule("toolbar");
        toolbar.addHandler("image", (value) => {
          this.uploadType = "image";
          if (value) {
            this.$refs.upload.$children[0].$refs.input.click();
          } else {
            this.quill.format("image", false);
          }
        });

        // 视频上传或链接
        toolbar.addHandler("video", async (value) => {
          const quill = this.Quill;
          const index = quill.getSelection().index;
          if (!value) return;

          const action = await this.$prompt("请选择输入方式", "插入视频", {
            distinguishCancelAndClose: true,
            confirmButtonText: "输入链接",
            cancelButtonText: "上传视频",
            inputPlaceholder: "请输入视频链接",
            inputValidator: (val) => {
              if (!val) return "请输入链接";
              if (!/^https?:\/\/.+\.(mp4|webm|ogg)(\?.*)?$/.test(val)) {
                return "请输入有效的视频链接 (mp4/webm/ogg)";
              }
              return true;
            },
          }).catch((err) => err);

          // 光标赋值
          this.videoIndex = index;

          if (action === "cancel") {
            // 上传本地视频
            this.uploadType = "video";
            this.$refs.upload.$children[0].$refs.input.click();
          } else if (action.value) {
            // 插入视频链接
            quill.insertEmbed(index, "video", action.value);
            quill.setSelection(index + 1);
          }
        });
      }
      this.Quill.pasteHTML(this.currentValue);
      // this.Quill.on("text-change", (delta, oldDelta, source) => {
      //   const html = this.$refs.editor.children[0].innerHTML;
      //   const text = this.Quill.getText();
      //   const quill = this.Quill;
      //   this.currentValue = html;
      //   this.$emit("input", html);
      //   this.$emit("on-change", { html, text, quill });
      // });
      this.Quill.on("text-change", () => {
        const contents = this.Quill.getContents();
        const html = this.getFullHtmlWithAlignment(contents);
        this.currentValue = html;
        this.$emit("input", html);
      });
      this.Quill.on("text-change", (delta, oldDelta, source) => {
        this.$emit("on-text-change", delta, oldDelta, source);
      });
      this.Quill.on("selection-change", (range, oldRange, source) => {
        this.$emit("on-selection-change", range, oldRange, source);
      });
      this.Quill.on("editor-change", (eventName, ...args) => {
        this.$emit("on-editor-change", eventName, ...args);
      });
    },
    getFullHtmlWithAlignment(delta) {
      const tempQuill = new Quill(document.createElement("div"));
      tempQuill.setContents(delta);

      // 处理所有图片
      const images = tempQuill.root.querySelectorAll("img");
      images.forEach((img) => {
        const blot = Quill.find(img);
        if (blot && blot.value().align) {
          const align = blot.value().align;
          img.style.cssText = ""; // 清除原有样式

          if (align === "left") {
            img.style.float = "left";
            img.style.margin = "0 1em 1em 0";
          } else if (align === "right") {
            img.style.float = "right";
            img.style.margin = "0 0 1em 1em";
          } else if (align === "center") {
            img.style.display = "block";
            img.style.margin = "auto";
          }
        }
      });

      return tempQuill.root.innerHTML;
    },
    // 上传前校检格式和大小
    handleBeforeUpload(file) {
      // 校检文件大小
      if (this.fileSize) {
        const isLt = file.size / 1024 / 1024 < this.fileSize;
        if (!isLt) {
          this.$message.error(`上传文件大小不能超过 ${this.fileSize} MB!`);
          return false;
        }
      }
      return true;
    },
    handleUploadSuccess(res, file) {
      // 获取富文本组件实例
      let quill = this.Quill;
      // 如果上传成功
      if (res.code == 200) {
        // 获取光标所在位置
        let index = quill.getSelection().index;
        if (this.uploadType === "image") {
          // 原方案
          // // 插入图片  res.url为服务器返回的图片地址
          // quill.insertEmbed(index, "image", res.data.url);
          // // 调整光标到最后
          // quill.setSelection(index + 1);

          const quill = this.Quill;
          const index = quill.getSelection().index;

          if (this.uploadType === "image") {
            // 插入图片并应用默认对齐方式
            quill.insertEmbed(
              index,
              "image",
              {
                url: res.data.url,
                align: "center", // 默认居中
              },
              Quill.sources.USER
            );

            quill.setSelection(index + 1);
          }
        } else if (this.uploadType === "video") {
          quill.insertEmbed(this.videoIndex, "video", res.data.url);
          // 调整光标到最后
          quill.setSelection(this.videoIndex + 1);
        }
      } else {
        this.$message.error(res.msg);
      }
    },
    handleUploadError() {
      if (this.uploadType === "image") {
        this.$message.error("图片插入失败");
      } else if (this.uploadType === "video") {
        this.$message.error("视频插入失败");
      }
    },
  },
};
</script>

<style>
.editor,
.ql-toolbar {
  white-space: pre-wrap !important;
  line-height: normal !important;
}
.quill-img {
  display: none;
}

.ql-snow .ql-tooltip[data-mode="link"]::before {
  content: "请输入链接地址:";
}
.ql-snow .ql-tooltip.ql-editing a.ql-action::after {
  border-right: 0px;
  content: "保存";
  padding-right: 0px;
}

.ql-snow .ql-tooltip[data-mode="video"]::before {
  content: "请输入视频地址:";
}

.ql-snow .ql-picker.ql-size .ql-picker-label::before,
.ql-snow .ql-picker.ql-size .ql-picker-item::before {
  content: "14px";
}
.ql-snow .ql-picker.ql-size .ql-picker-label[data-value="small"]::before,
.ql-snow .ql-picker.ql-size .ql-picker-item[data-value="small"]::before {
  content: "10px";
}
.ql-snow .ql-picker.ql-size .ql-picker-label[data-value="large"]::before,
.ql-snow .ql-picker.ql-size .ql-picker-item[data-value="large"]::before {
  content: "18px";
}
.ql-snow .ql-picker.ql-size .ql-picker-label[data-value="huge"]::before,
.ql-snow .ql-picker.ql-size .ql-picker-item[data-value="huge"]::before {
  content: "32px";
}

.ql-snow .ql-picker.ql-header .ql-picker-label::before,
.ql-snow .ql-picker.ql-header .ql-picker-item::before {
  content: "文本";
}
.ql-snow .ql-picker.ql-header .ql-picker-label[data-value="1"]::before,
.ql-snow .ql-picker.ql-header .ql-picker-item[data-value="1"]::before {
  content: "标题1";
}
.ql-snow .ql-picker.ql-header .ql-picker-label[data-value="2"]::before,
.ql-snow .ql-picker.ql-header .ql-picker-item[data-value="2"]::before {
  content: "标题2";
}
.ql-snow .ql-picker.ql-header .ql-picker-label[data-value="3"]::before,
.ql-snow .ql-picker.ql-header .ql-picker-item[data-value="3"]::before {
  content: "标题3";
}
.ql-snow .ql-picker.ql-header .ql-picker-label[data-value="4"]::before,
.ql-snow .ql-picker.ql-header .ql-picker-item[data-value="4"]::before {
  content: "标题4";
}
.ql-snow .ql-picker.ql-header .ql-picker-label[data-value="5"]::before,
.ql-snow .ql-picker.ql-header .ql-picker-item[data-value="5"]::before {
  content: "标题5";
}
.ql-snow .ql-picker.ql-header .ql-picker-label[data-value="6"]::before,
.ql-snow .ql-picker.ql-header .ql-picker-item[data-value="6"]::before {
  content: "标题6";
}

.ql-snow .ql-picker.ql-font .ql-picker-label::before,
.ql-snow .ql-picker.ql-font .ql-picker-item::before {
  content: "标准字体";
}
.ql-snow .ql-picker.ql-font .ql-picker-label[data-value="serif"]::before,
.ql-snow .ql-picker.ql-font .ql-picker-item[data-value="serif"]::before {
  content: "衬线字体";
}
.ql-snow .ql-picker.ql-font .ql-picker-label[data-value="monospace"]::before,
.ql-snow .ql-picker.ql-font .ql-picker-item[data-value="monospace"]::before {
  content: "等宽字体";
}

/* 调整手柄样式 */
.ql-blot-formatter__resize-handle {
  width: 10px;
  height: 10px;
  background-color: #4285f4;
  border: 2px solid white;
  border-radius: 50%;
}

/* 对齐工具栏样式 */
.ql-blot-formatter__toolbar {
  background: #fff;
  border: 1px solid #ddd;
  border-radius: 4px;
  box-shadow: 0 2px 4px rgba(0, 0, 0, 0.2);
  padding: 4px;
}

/* 对齐按钮样式 */
.ql-blot-formatter__toolbar-button {
  width: 24px;
  height: 24px;
  padding: 3px;
  margin: 0 2px;
  cursor: pointer;
}

/* 图片对齐样式 */
.ql-editor img.align-left {
  display: inline;
  float: left;
  margin-right: 1em;
}

.ql-editor img.align-center {
  display: block;
  margin: 0 auto;
}

.ql-editor img.align-right {
  display: inline;
  float: right;
  margin-left: 1em;
}

/* 强制保留对齐样式 */
.ql-editor img[style*="float:left"],
.ql-editor img[style*="float: left"] {
  float: left !important;
  margin: 0 1em 1em 0 !important;
}

.ql-editor img[style*="float:right"],
.ql-editor img[style*="float: right"] {
  float: right !important;
  margin: 0 0 1em 1em !important;
}

.ql-editor img[style*="margin: 0 auto"],
.ql-editor img[style*="margin:0 auto"] {
  display: block !important;
  margin: 0 auto 1em !important;
  float: none !important;
}
</style>
